numpy
scikit-learn
pytorch::pytorch 
pytorch::torchvision
tqdm
